package com.zhentao.config;

import com.zhentao.websocket.ChatEndpoint;
import org.springframework.boot.ApplicationRunner;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * WebSocket初始化配置
 * 确保在Spring容器启动完成后初始化WebSocket相关服务
 */
@Configuration
public class WebSocketInitializer {

    /**
     * 应用启动后初始化WebSocket服务
     */
    @Bean
    public ApplicationRunner initWebSocketServices() {
        return args -> {
            // 延迟一点时间确保所有Bean都已初始化
            Thread.sleep(1000);
            
            // 初始化ChatEndpoint的静态服务
            ChatEndpoint.initializeStaticServices();
            
            System.out.println("WebSocket服务初始化完成");
        };
    }
}
